package com.qjc.demo.sort.快速排序;

import java.util.Arrays;

/**
 * @ClassName: QuickSort1
 * @Description: 快速排序-双边循环法
 * @Author: qjc
 * @Date: 2022/5/2 11:41 AM
 */
public class QuickSort1 {

    public static void quickSort(int[] nums, int start, int end) {
        if (start >= end) return;
        // 得到基准元素位置
        int pivotIndex = getPivot(nums, start, end);
        // 对左边进行排序
        quickSort(nums, start, pivotIndex - 1);
        // 对右边进行排序
        quickSort(nums, pivotIndex + 1, end);
    }

    // 分治（双边循环法）：得到基准位置
    private static int getPivot(int[] nums, int start, int end) {
        // 选取第一个位置（也可以随机选择）的元素作为基准元素
        int pivot = nums[start];
        int left = start;
        int right = end;
        while (left != right) {
            while (left < right && nums[right] > pivot) {
                // 如果右指针所在位置的元素大于基准元素，则右指针左移
                right--;
            }
            while (left < right && nums[left] <= pivot) {
                // 如果左指针所在位置的元素小于基准元素，则左指针右移
                left++;
            }
            if (left < right) {
                // 否则，进行交换
                int tmp = nums[left];
                nums[left] = nums[right];
                nums[right] = tmp;
            }
        }
        // 如果左右指针重合，则基准元素和重合元素进行交换
        nums[start] = nums[left];
        nums[left] = pivot;
        return left;
    }

    public static void main(String[] args) {
        int[] arr = new int[]{4, 7, 3, 5, 6, 2, 8, 1};
        quickSort(arr, 0, arr.length - 1);
        System.out.println(Arrays.toString(arr));
    }

}
